from smac.env import StarCraft2Env
import numpy as np

def make_sc2_env(map_name="2s3z", debug=False):
    """创建SMAC环境并返回封装后的接口"""
    env = StarCraft2Env(
        map_name=map_name,
        debug=debug,
        reward_only_positive=False,
        obs_last_action=False
    )
    env.reset()
    return env

def get_obs_size(env):
    """获取观察空间维度"""
    return env.get_obs_size()

def get_state_size(env):
    """获取全局状态维度"""
    return env.get_state_size()